how to draw square in python